package com.treeroot.sparsearray;

/**
 * 稀疏数组稀疏数组
 * @author gpt优化的
 * @describe :
 * 将二维数组转稀疏数组:
 * 遍历原始的二维数组，得到非0数据的个数 sum
 * 根据 sum 创建稀疏数组 sparseArr int[sum+1][3]，第一行分别存储原始数组的行数、列数与非0数据的个数
 * 将二维数组的有效数据存入稀疏数组中
 * 将稀疏数组恢复成原始二维数组:
 * 先读取稀疏数组的第一行，根据第一行的数据创建原始的二维数组
 * 读取稀疏数组后几行的数据，并赋给原始的二维数组
 */
public class sparseArrayByGPT {


    /**
     * 将二维数组转稀疏数组
     *
     * @param chessArr 二维数组
     * @return 返回稀疏数组
     */
    public static int[][] toSparseArray(int[][] chessArr) {
        // 计算非0元素的个数
        int sum = 0;
        for (int[] row : chessArr) {
            for (int data : row) {
                if (data != 0) {
                    sum++;
                }
            }
        }

        // 创建稀疏数组
        int[][] sparseArr = new int[sum + 1][3];
        sparseArr[0][0] = chessArr.length;
        sparseArr[0][1] = chessArr[0].length;
        sparseArr[0][2] = sum;

        // 存储有效数据
        int count = 0;
        for (int i = 0; i < chessArr.length; i++) {
            for (int j = 0; j < chessArr[0].length; j++) {
                if (chessArr[i][j] != 0) {
                    count++;
                    sparseArr[count][0] = i;
                    sparseArr[count][1] = j;
                    sparseArr[count][2] = chessArr[i][j];
                }
            }
        }

        return sparseArr;
    }

    /**
     * 将稀疏数组恢复成原始二维数组
     *
     * @param sparseArr 稀疏数组
     * @return 返回原始的二维数组
     */
    public static int[][] toChessArr(int[][] sparseArr) {
        int[][] chessArr = new int[sparseArr[0][0]][sparseArr[0][1]];
        for (int i = 1; i < sparseArr.length; i++) {
            chessArr[sparseArr[i][0]][sparseArr[i][1]] = sparseArr[i][2];
        }
        return chessArr;
    }

    public static void main(String[] args) {
        int[][] chessArr1 = new int[11][11];
        chessArr1[1][2] = 1;
        chessArr1[2][3] = 2;

        System.out.println("原始二维数组chessArr1");
        for (int[] row : chessArr1) {
            for (int data : row) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }

        int[][] sparseArr = toSparseArray(chessArr1);
        System.out.println("得到的稀疏数组如下:");
        for (int[] rows : sparseArr) {
            System.out.printf("%d\t%d\t%d\t\n", rows[0], rows[1], rows[2]);
        }

        int[][] chessArr2 = toChessArr(sparseArr);
        System.out.println("恢复后的二维数组chessArr2");
        for (int[] row : chessArr2) {
            for (int data : row) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
    }
}
